#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <time.h>
#include <stdlib.h>
#include <unistd.h>
// #include "Windows.h"

#include "HCNetSDK.h"
#include "PlayM4.h"
#include <opencv2/opencv.hpp>
#include "opencv2/face.hpp"
#include "cv.h"
#include "highgui.h"

#define USECOLOR 1
using namespace cv;
using namespace cv::face;
using namespace std;

//--------------------------------------------
int iPicNum = 0; //Set channel NO.
LONG nPort = -1;
HWND hWnd = 0;
String face_cascade_name = "/usr/local/opencv/share/OpenCV/haarcascades/haarcascade_frontalface_alt.xml";
// String eyes_cascade_name = "haarcascade_eye_tree_eyeglasses.xml";
CascadeClassifier face_cascade;
vector<Rect> faces; //存储脸数据

void yv12toYUV(char *outYuv, char *inYv12, int width, int height, int widthStep)
{
    int col, row;
    unsigned int Y, U, V;
    int tmp;
    int idx;

    //printf("widthStep=%d.\n",widthStep);

    for (row = 0; row < height; row++)
    {
        idx = row * widthStep;
        int rowptr = row * width;

        for (col = 0; col < width; col++)
        {
            //int colhalf=col>>1;
            tmp = (row / 2) * (width / 2) + (col / 2);
            //         if((row==1)&&( col>=1400 &&col<=1600))
            //         {
            //          printf("col=%d,row=%d,width=%d,tmp=%d.\n",col,row,width,tmp);
            //          printf("row*width+col=%d,width*height+width*height/4+tmp=%d,width*height+tmp=%d.\n",row*width+col,width*height+width*height/4+tmp,width*height+tmp);
            //         }
            Y = (unsigned int)inYv12[row * width + col];
            U = (unsigned int)inYv12[width * height + width * height / 4 + tmp];
            V = (unsigned int)inYv12[width * height + tmp];
            //         if ((col==200))
            //         {
            //         printf("col=%d,row=%d,width=%d,tmp=%d.\n",col,row,width,tmp);
            //         printf("width*height+width*height/4+tmp=%d.\n",width*height+width*height/4+tmp);
            //         return ;
            //         }
            if ((idx + col * 3 + 2) > (1200 * widthStep))
            {
                //printf("row * widthStep=%d,idx+col*3+2=%d.\n",1200 * widthStep,idx+col*3+2);
            }
            outYuv[idx + col * 3] = Y;
            outYuv[idx + col * 3 + 1] = U;
            outYuv[idx + col * 3 + 2] = V;
        }
    }
    //printf("col=%d,row=%d.\n",col,row);
}

//解码回调 视频为YUV数据(YV12)，音频为PCM数据
void CALLBACK DecCBFun(int nPort, char *pBuf, int nSize, FRAME_INFO *pFrameInfo, void *nReserved1, int nReserved2)
{
    long lFrameType = pFrameInfo->nType;
    if (!face_cascade.load(face_cascade_name))
    {
        //加载脸部分类器失败！
        return;
    }

    if (lFrameType == T_YV12)
    {
#if USECOLOR
        // int start = clock();
        IplImage *pImgYCrCb = cvCreateImage(cvSize(pFrameInfo->nWidth, pFrameInfo->nHeight), 8, 3);           //得到图像的Y分量
        yv12toYUV(pImgYCrCb->imageData, pBuf, pFrameInfo->nWidth, pFrameInfo->nHeight, pImgYCrCb->widthStep); //得到全部RGB图像
        IplImage *pImg = cvCreateImage(cvSize(pFrameInfo->nWidth, pFrameInfo->nHeight), 8, 3);
        cvCvtColor(pImgYCrCb, pImg, CV_YCrCb2RGB);
        // int end = clock();
        // printf("使用yv12toYUV \n");
#else
        IplImage *pImg = cvCreateImage(cvSize(pFrameInfo->nWidth, pFrameInfo->nHeight), 8, 1);
        memcpy(pImg->imageData, pBuf, pFrameInfo->nWidth * pFrameInfo->nHeight);
        // printf("没有使用yv12toYUV \n");
#endif

        // printf("用时：%d\n",end-start);
        IplImage *image_gray = cvCreateImage(cvSize(pFrameInfo->nWidth, pFrameInfo->nHeight), 8, 1);
        cvCvtColor(pImg, image_gray, CV_RGB2GRAY);//转为灰度图
        //检测关于脸部位置
        Mat mtx = cvarrToMat(image_gray);
        Mat mtxs = cvarrToMat(pImg);
        cv::equalizeHist(mtx, mtx);//直方图均衡化，增加对比度方便处理
        // // Mat mtx(image_gray, true);
        face_cascade.detectMultiScale(mtx, faces, 1.1, 2, 0 | CV_HAAR_SCALE_IMAGE, Size(30, 30));//检测
        for (size_t i = 0; i < faces.size(); i++)
        {   
            cv::rectangle(mtxs, faces[i], Scalar(0, 0, 255));      //用矩形画出检测到的位置
        }
        // cvShowImage("IPCamera", pImg);
        imshow("IPCamera", mtxs);
        cvWaitKey(30);
#if USECOLOR
        cvReleaseImage(&pImgYCrCb);
        cvReleaseImage(&pImg);
#else
        cvReleaseImage(&pImg);
#endif
        cvReleaseImage(&image_gray);
        //此时是YV12格式的视频数据，保存在pBuf中，可以fwrite(pBuf,nSize,1,Videofile);
        //fwrite(pBuf,nSize,1,fp);
    }
    /***************
	else if (lFrameType ==T_AUDIO16)
	{
		//此时是音频数据，数据保存在pBuf中，可以fwrite(pBuf,nSize,1,Audiofile);
		
	}
	else
	{

	}
	*******************/
}

///实时流回调
void CALLBACK fRealDataCallBack(LONG lRealHandle, DWORD dwDataType, BYTE *pBuffer, DWORD dwBufSize, void *pUser)
{
    DWORD dRet;
    switch (dwDataType)
    {
    case NET_DVR_SYSHEAD: //系统头
        // cout << "系统头" << endl;
        if (!PlayM4_GetPort(&nPort)) //获取播放库未使用的通道号
        {
            break;
        }
        if (dwBufSize > 0)
        {
            if (!PlayM4_OpenStream(nPort, pBuffer, dwBufSize, 1024 * 1024))
            {
                dRet = PlayM4_GetLastError(nPort);
                break;
            }
            //设置解码回调函数 只解码不显示
            if (!PlayM4_SetDecCallBack(nPort, DecCBFun))
            {
                dRet = PlayM4_GetLastError(nPort);
                break;
            }

            //设置解码回调函数 解码且显示
            // if (!PlayM4_SetDecCallBackEx(nPort, DecCBFun, NULL, NULL))
            // {
            //     dRet = PlayM4_GetLastError(nPort);
            //     break;
            // }

            //打开视频解码
            if (!PlayM4_Play(nPort, hWnd))
            {
                dRet = PlayM4_GetLastError(nPort);
                break;
            }

            //打开音频解码, 需要码流是复合流
            if (!PlayM4_PlaySound(nPort))
            {
                dRet = PlayM4_GetLastError(nPort);
                break;
            }
        }
        break;

    case NET_DVR_STREAMDATA: //码流数据
        // cout << "码流数据" << endl;
        if (dwBufSize > 0 && nPort != -1)
        {
            BOOL inData = PlayM4_InputData(nPort, pBuffer, dwBufSize);
            while (!inData)
            {
                sleep(10);
                inData = PlayM4_InputData(nPort, pBuffer, dwBufSize);
                // OutputDebugString(L"PlayM4_InputData failed \n");
            }
        }
        break;
    }
}

void CALLBACK g_ExceptionCallBack(DWORD dwType, LONG lUserID, LONG lHandle, void *pUser)
{
    char tempbuf[256] = {0};
    switch (dwType)
    {
    case EXCEPTION_RECONNECT: //预览时重连
        printf("----------reconnect--------%d\n", time(NULL));
        break;
    default:
        break;
    }
}

int main()
{

    //---------------------------------------
    // 初始化
    NET_DVR_Init();
    //设置连接时间与重连时间
    NET_DVR_SetConnectTime(2000, 1);
    NET_DVR_SetReconnect(10000, true);

    //---------------------------------------
    // 获取控制台窗口句柄
    //HMODULE hKernel32 = GetModuleHandle((LPCWSTR)"kernel32");
    //GetConsoleWindow = (PROCGETCONSOLEWINDOW)GetProcAddress(hKernel32,"GetConsoleWindow");

    //---------------------------------------
    // 注册设备
    LONG lUserID;
    NET_DVR_DEVICEINFO_V30 struDeviceInfo;
    lUserID = NET_DVR_Login_V30("192.168.0.71", 8000, "admin", "12345", &struDeviceInfo);
    if (lUserID < 0)
    {
        printf("Login error, %d\n", NET_DVR_GetLastError());
        NET_DVR_Cleanup();
        return -1;
    }

    //---------------------------------------
    //设置异常消息回调函数
    NET_DVR_SetExceptionCallBack_V30(0, NULL, g_ExceptionCallBack, NULL);

    cvNamedWindow("IPCamera");
    //---------------------------------------
    //启动预览并设置回调数据流
    // NET_DVR_CLIENTINFO ClientInfo;
    // ClientInfo.lChannel = 1;  //Channel number 设备通道号
    // ClientInfo.hPlayWnd = 0;  //窗口为空，设备SDK不解码只取流
    // ClientInfo.lLinkMode = 0; //Main Stream
    // ClientInfo.sMultiCastIP = NULL;

    NET_DVR_PREVIEWINFO struPlayInfo = {0};
    struPlayInfo.hPlayWnd = 0;     //需要 SDK 解码时句柄设为有效值,仅取流不解码时可设为空
    struPlayInfo.lChannel = 1;     //预览通道号
    struPlayInfo.dwStreamType = 0; //0-主码流,1-子码流,2-码流 3,3-码流 4,以此类推
    struPlayInfo.dwLinkMode = 0;   //0- TCP 方式,1- UDP 方式,2- 多播方式,3- RTP 方式,4-RTP/RTSP,5-RSTP/HTTP
    struPlayInfo.bBlocked = 1;     //0- 非阻塞取流,1- 阻塞取流

    LONG lRealPlayHandle;
    lRealPlayHandle = NET_DVR_RealPlay_V40(lUserID, &struPlayInfo, fRealDataCallBack, NULL);
    if (lRealPlayHandle < 0)
    {
        printf("NET_DVR_RealPlay_V30 failed! Error number: %d\n", NET_DVR_GetLastError());
        return -1;
    }

    cvWaitKey(0);
    //   Sleep(-1);

    //fclose(fp);
    //---------------------------------------
    //关闭预览
    if (!NET_DVR_StopRealPlay(lRealPlayHandle))
    {
        printf("NET_DVR_StopRealPlay error! Error number: %d\n", NET_DVR_GetLastError());
        return -1;
    }
    //注销用户
    NET_DVR_Logout(lUserID);
    NET_DVR_Cleanup();

    return 0;
}
